package lnet

import (
	"bytes"
	"encoding/binary"
	"errors"
	"linx/liface"
	"linx/utils"
)

/*
	封包&拆包 模块
	直接面向TCP連接中的数据流，用于处理TCP战报问题
*/

type DataPck struct{}

// 拆包封包实例的一个初始化方法
func NewDataPack() *DataPck {
	return &DataPck{}
}

// 获取包的头的长度方法
func (d *DataPck) GetHeadLen() uint32 {
	// DataLen uint32（4字节）+ ID uint32 (4字节)
	return 8
}

// 封包方法 msgLen msgID data
func (d *DataPck) Pack(msg liface.IMessage) ([]byte, error) {
	// 创建一个存放bytes 字节的缓冲区
	dataBuff := bytes.NewBuffer([]byte{})
	// 将dataLen写进 databuff 中
	if err := binary.Write(dataBuff, binary.LittleEndian, msg.GetMsgLen()); err != nil {
		return nil, err
	}

	// 将MsgId 写进databuff中
	if err := binary.Write(dataBuff, binary.LittleEndian, msg.GetMsgId()); err != nil {
		return nil, err
	}

	// 将data数据 写进databuff中
	if err := binary.Write(dataBuff, binary.LittleEndian, msg.GetData()); err != nil {
		return nil, err
	}

	return dataBuff.Bytes(), nil
}

// 拆包方法 将包的 Head 信息读出来 再根据 Head 信息里的 dataLen 长度再读一次
func (d *DataPck) Unpack(binaryData []byte) (liface.IMessage, error) {
	// 创建一个从输入二进制数据的 ioReader
	reader := bytes.NewReader(binaryData)

	// 只解压 head 信息 得到 len 和 id
	msg := &Message{}

	// 读 len
	if err := binary.Read(reader, binary.LittleEndian, &msg.DataLen); err != nil {
		return nil, err
	}
	// 读 id
	if err := binary.Read(reader, binary.LittleEndian, &msg.Id); err != nil {
		return nil, err
	}
	// 判断datalen是否已经超出允许的最大包长度
	if utils.GlobalObject.MaxPackageSize >0 && msg.DataLen > utils.GlobalObject.MaxPackageSize {
		return nil, errors.New("too large msg data recv!")
	}

	return msg, nil
}
